Pytorch – 检测CUDA、cuDNN以及GPU版本的Pytorch是否安装成功、GPU显存测试
1 测试代码 import time import torch if __name__ == '__main__': print('torch版本：'+torch.__version__) print('cuda是否可用：'+str(torch.cuda.is_available())) print…
- Pytorch
- 2021-08-06
Python – Linux/Centos/Ubuntu查看CUDA/cuDNN版本号
1 查看CUDA版本号 低版本使用： nvcc -v 高版本使用： nvcc --version 查看CUDA版本号。 2 查看cuDNN版本号 旧版本cuDNN使用以下命令查看版本号： cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -…
- Python
- 2021-06-11